/*
 * Copyright (c) 2021 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.sleepbot.datetimepicker.time;

import com.fourmob.datetimepicker.ResourceTable;

import ohos.agp.components.Component;
import ohos.agp.render.Canvas;
import ohos.agp.render.Paint;
import ohos.agp.utils.Color;
import ohos.agp.utils.TextAlignment;
import ohos.app.Context;
import ohos.global.resource.ResourceManager;
import ohos.hiviewdfx.HiLog;
import ohos.hiviewdfx.HiLogLabel;

import java.text.DateFormatSymbols;

/**
 * @since 2021-03-25
 */
public class AmPmCirclesView extends Component implements Component.DrawTask {
    private static final String TAG = "AmPmCirclesView";

    private static HiLogLabel label = new HiLogLabel(HiLog.LOG_APP, 0x000110, "DateTimePicker");

    // Alpha level of blue color for selected circle.
    private static final int SELECTED_ALPHA = 51;

    // Alpha level of blue color for pressed circle.
    private static final int PRESSED_ALPHA = 175;
    private static final int AM = TimePickerDialog.AM;
    private static final int PM = TimePickerDialog.PM;

    private final Paint mPaint = new Paint();
    private int mWhite;
    private int mAmPmTextColor;
    private int mBlue;
    private float mCircleRadiusMultiplier;
    private float mAmPmCircleRadiusMultiplier;
    private String mAmText;
    private String mPmText;
    private boolean mIsInitialized;

    private boolean mDrawValuesReady;
    private int mAmPmCircleRadius;
    private int mAmXCenter;
    private int mPmXCenter;
    private int mAmPmYCenter;
    private int mAmOrPm;
    private int mAmOrPmPressed;

    public AmPmCirclesView(Context context) {
        super(context);
        mIsInitialized = false;
    }

    public void initialize(Context context, int amOrPm) {
        addDrawTask(this::onDraw);
        if (mIsInitialized) {
            HiLog.error(label, "AmPmCirclesView may only be initialized once.");
            return;
        }

        try {
            ResourceManager resourceManager = context.getResourceManager();
            mWhite = resourceManager.getElement(ResourceTable.Color_white).getColor();
            mAmPmTextColor = resourceManager.getElement(ResourceTable.Color_ampm_text_color).getColor();
            mBlue = resourceManager.getElement(ResourceTable.Color_light_blue).getColor();
            mPaint.setAntiAlias(true);
            mPaint.setTextAlign(TextAlignment.CENTER);
            mCircleRadiusMultiplier = Float.parseFloat(
                resourceManager.getElement(ResourceTable.String_circle_radius_multiplier).getString());
            mAmPmCircleRadiusMultiplier = Float.parseFloat(
                resourceManager.getElement(ResourceTable.String_ampm_circle_radius_multiplier).getString());
        } catch (Exception e) {
            e.printStackTrace();
        }

        String[] amPmTexts = new DateFormatSymbols().getAmPmStrings();
        mAmText = amPmTexts[0]; // mAmText = "上午";
        mPmText = amPmTexts[1]; // mPmText = "下午";

        setAmOrPm(amOrPm);
        mAmOrPmPressed = -1;
        mIsInitialized = true;
    }

    public void setAmOrPm(int amOrPm) {
        mAmOrPm = amOrPm;
    }

    public void setAmOrPmPressed(int amOrPmPressed) {
        mAmOrPmPressed = amOrPmPressed;
    }

    /**
     * Calculate whether the coordinates are touching the AM or PM circle.
     *
     * @param xCoord
     * @param yCoord
     * @return int类型
     */
    public int getIsTouchingAmOrPm(float xCoord, float yCoord) {
        if (!mDrawValuesReady) {
            return -1;
        }

        int squaredYDistance = (int) ((yCoord - mAmPmYCenter) * (yCoord - mAmPmYCenter));

        int distanceToAmCenter =
            (int) Math.sqrt((xCoord - mAmXCenter) * (xCoord - mAmXCenter) + squaredYDistance);
        if (distanceToAmCenter <= mAmPmCircleRadius) {
            return AM;
        }

        int distanceToPmCenter =
            (int) Math.sqrt((xCoord - mPmXCenter) * (xCoord - mPmXCenter) + squaredYDistance);
        if (distanceToPmCenter <= mAmPmCircleRadius) {
            return PM;
        }
        return -1;
    }

    @Override
    public void onDraw(Component component, Canvas canvas) {
        int viewWidth = getWidth();
        if (viewWidth == 0 || !mIsInitialized) {
            return;
        }

        if (!mDrawValuesReady) {
            int layoutXCenter = getWidth() / 2;
            int layoutYCenter = getHeight() / 2;
            int circleRadius =
                (int) (Math.min(layoutXCenter, layoutYCenter) * mCircleRadiusMultiplier);
            mAmPmCircleRadius = (int) (circleRadius * mAmPmCircleRadiusMultiplier);
            int textSize = mAmPmCircleRadius * 3 / 4;
            mPaint.setTextSize(textSize);

            // Line up the vertical center of the AM/PM circles with the bottom of the main circle.
            mAmPmYCenter = layoutYCenter - mAmPmCircleRadius / 2 + circleRadius;
            // Line up the horizontal edges of the AM/PM circles with the horizontal edges
            // of the main circle.
            mAmXCenter = layoutXCenter - circleRadius + mAmPmCircleRadius;
            mPmXCenter = layoutXCenter + circleRadius - mAmPmCircleRadius;

            mDrawValuesReady = true;
        }

        // We'll need to draw either a lighter blue (for selection), a darker blue (for touching)
        // or white (for not selected).
        int amColor = mWhite;
        int amAlpha = 255;
        int pmColor = mWhite;
        int pmAlpha = 255;
        if (mAmOrPm == AM) {
            amColor = mBlue;
            amAlpha = SELECTED_ALPHA;
        } else if (mAmOrPm == PM) {
            pmColor = mBlue;
            pmAlpha = SELECTED_ALPHA;
        }
        if (mAmOrPmPressed == AM) {
            amColor = mBlue;
            amAlpha = PRESSED_ALPHA;
        } else if (mAmOrPmPressed == PM) {
            pmColor = mBlue;
            pmAlpha = PRESSED_ALPHA;
        }

        // Draw the two circles.
        mPaint.setColor(new Color(amColor));

        canvas.drawCircle(mAmXCenter, mAmPmYCenter, mAmPmCircleRadius, mPaint);
        mPaint.setColor(new Color(pmColor));

        canvas.drawCircle(mPmXCenter, mAmPmYCenter, mAmPmCircleRadius, mPaint);

        // Draw the AM/PM texts on top.
        mPaint.setColor(new Color(mAmPmTextColor));
        int textYCenter = mAmPmYCenter - (int) (mPaint.descent() + mPaint.ascent()) / 2;
        canvas.drawText(mPaint, mAmText, mAmXCenter, textYCenter);
        canvas.drawText(mPaint, mPmText, mPmXCenter, textYCenter);
    }
}
